"""
# -*- coding: utf-8 -*-
# @Time    : 2023/3/4 16:25
# @Author  : 王摇摆
# @FileName: argmax.py
# @Software: PyCharm
# @Blog    ：https://blog.csdn.net/weixin_44943389?type=blog
"""
import numpy    as  np

A = np.array([[1, 2, 3],
              [4, 5, 6]])

# 从列的方向找行元素中的最大值
print(np.argmax(a=A, axis=0))

# 从行的方向找列元素中的最大值
print(np.argmax(a=A, axis=1))
